package com.seanLab.tool.TagSuggestModel;

import com.seanLab.domain.TagInfo;

import java.util.ArrayList;
import java.util.List;

/***
 * Only return tag extracted from content
 */
public class ModelKernelStrategyBasic implements ModelKernelStrategy {
    @Override
    public List<TagInfo> doSuggest(List<List<TagInfo>> tagsList, RawInfoType[] rawInfoTypes) {
        for (int i = 0; i < rawInfoTypes.length; i++) {
            if (rawInfoTypes[i] == RawInfoType.CONTENT) {
                return tagsList.get(i);
            }
        }
        return new ArrayList<>();
    }
}
